# -*- coding: utf-8 -*-
# windows 下如果出现编码问题，将 utf-8 改为 cp936
import os

def findfile(key, inputdir='.'):
    found_list = []
    all_files = os.listdir(inputdir)
    for name in all_files:  # 遍历目录下的所有文件
        full_name = inputdir + '/' + name
        if key in name:  # 如果文件名中有关键字
            found_list.append(full_name)
        else:
            try:  # 加上 try，避免打开非文本文件时报错退出
                with open(full_name) as f:
                    for l in f:  # 对每一行进行遍历，效果等同于 for l in f.readlines() 但效率更高
                        if key in l:  # 如果当前行中有关键字，就加入列表并跳出此文件
                            found_list.append(full_name + ' : ' + l.strip())
                            break
            except:
                pass
    return found_list

    # 如果需查找包括子目录中的所有文件，要使用 os.walk
    # os.walk 获取指定目录下的所有的目录和文件列表（包括子目录）
    # 相应的 for 循环也需要改动，需使用两层循环
    # for path, dirnames, filenames in os.walk(inputdir):  # 遍历目录下的所有子目录
    #     for name in filenames:  # 遍历每个子目录中的文件
    #         full_name = path + '/' + name
    #         ... （后面代码同上）


# 输入搜索关键字和路径
keyword = input('匹配关键字：')
path = input('搜索目录（不填默认为当前目录）：')
if not path.strip():  # 如果没有有效字符
    path = '.'   # . 表示当前目录

result = findfile(keyword, path)

# 输出结果
print('========== 匹配结果 ===========')
for r in result:
    print(r)